#-*-Python-*-
import tf_agents.environments.suite_bsuite

## Configure Environment
ENVIRONMENT = @suite_bsuite.load()
suite_bsuite.load.bsuite_id = %BSUITE_ID
suite_bsuite.load.record = %RECORD
suite_bsuite.load.save_path = %SAVE_PATH
suite_bsuite.load.logging_mode = %LOGGING_MODE
# Note: The BSUITE_ID can be overridden by passing the command line flag:
# --gin_param="BSUITE_ID='deep_sea/0'". Same for SAVE_PATH and LOGGING_MODE.
BSUITE_ID = 'deep_sea/0'
RECORD = True
SAVE_PATH = None
LOGGING_MODE = 'terminal'
